@echo off

REM Qwen3-0.6B 模型部署辅助脚本
REM 此脚本将帮助您在Windows环境中部署Qwen3-0.6B模型

echo ====================================
echo Qwen3-0.6B 模型部署辅助脚本
echo ====================================
echo.

REM 检查Python版本
echo 正在检查Python版本...
python --version

REM 创建虚拟环境
echo.
echo 创建虚拟环境...
python -m venv qwen3_env

REM 激活虚拟环境
echo.
echo 激活虚拟环境...
call qwen3_env\Scripts\activate

REM 升级pip
echo.
echo 升级pip...
pip install --upgrade pip

REM 安装PyTorch（CPU版本）
echo.
echo 安装PyTorch（CPU版本）...
pip install torch torchvision torchaudio

REM 安装Transformers和其他依赖
echo.
echo 安装Transformers和其他依赖...
pip install transformers>=4.51.0 accelerate

echo.
echo ====================================
echo 环境配置完成！
echo 请按照以下步骤操作：
echo 1. 确保您的Python版本在3.9-3.12之间（推荐3.10或3.11）
echo 2. 虚拟环境已创建并激活

echo 3. 若需使用GPU，请手动安装对应版本的PyTorch:

echo    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
echo.
echo 4. 运行以下命令启动Qwen3-0.6B模型:
echo    python index.py
echo.
echo 5. 若遇到Python版本兼容性问题，请安装Python 3.10或3.11
echo ====================================

echo.
echo 按任意键退出...
pause > nul